from enum import Enum

TRAIN_PATH = "dataset/train.txt"
TEST_PATH = "dataset/test.txt"
NORMAL_MODEL = ['alexnet', 'vgg', 'mobilenet']
ATTENTION_MODEL = ['resnet1', 'resnet8']
MODEL_TYPE = ['alexnet', 'vgg', 'mobilenet', 'resnet_1', 'resnet_8']
MODEL_FUNC = ['origin', 'kl', 'attn', 'kl_attn']
PATH_GENERATE_TYPE = ['single_same', 'multi_same', 'multi_diff']

TIME_MARK_PATH = 'time_mark.dp'
TOTAL_MARK_PATH = 'dataset/total_time_mark.dp'

class AlgoType(Enum):
    SINGLE_DOPPLER = 0
    MULTI_DOPPLER = 1

class DataModoType(Enum):
    SINGLE = 0
    MULTIPLE = 1